@echo off
chcp 65001 >nul
echo ========================================
echo CreateDocument API 测试套件
echo ========================================

cd /d "%~dp0\.."

echo.
echo [1/3] 运行服务层测试...
echo ----------------------------------------
python -m pytest tests/test_knowledge_service.py -k "create_document" -v --tb=short
if errorlevel 1 (
    echo.
    echo [FAIL] 服务层测试失败
    goto :error
)

echo.
echo [2/3] 运行路由层测试...
echo ----------------------------------------
python -m pytest tests/test_knowledge_router.py -k "create_document" -v --tb=short
if errorlevel 1 (
    echo.
    echo [FAIL] 路由层测试失败
    goto :error
)

echo.
echo [3/3] 检查服务器状态...
echo ----------------------------------------
curl -s http://localhost:8000/health >nul 2>&1
if errorlevel 1 (
    echo [警告] 服务器未运行，跳过真实环境测试
    echo.
    echo 如需运行真实环境测试，请在另一个终端启动服务器:
    echo   python src/main.py
    echo.
    echo 然后运行:
    echo   python tests/test_create_document_real.py
    goto :success
)

echo [OK] 服务器正在运行
echo.
echo 运行真实环境测试...
python tests/test_create_document_real.py
if errorlevel 1 (
    echo.
    echo [FAIL] 真实环境测试失败
    goto :error
)

:success
echo.
echo ========================================
echo [SUCCESS] 所有测试通过！
echo ========================================
echo.
echo 测试总结:
echo   - 服务层测试: PASSED
echo   - 路由层测试: PASSED
if not errorlevel 1 echo   - 真实环境测试: PASSED
echo.
pause
exit /b 0

:error
echo.
echo ========================================
echo [FAIL] 测试失败！
echo ========================================
echo.
pause
exit /b 1
